本次主題是以colab的環境進行學習的,在本篇文章中,我將講解影像辨識的基礎技能在接下來的文章中這些技能將多次出現,先讀過這些語法再繼續去看後面的文章會比較能快速上手喔。依照進度每個禮拜都會記錄不同的影像辨識方法,基本順序會從:
在我們上一篇文章中我們已經把資料集準備好了,接下來我們就回到colab裡面進行模型訓練吧。
雲端硬碟掛載:
from google.colab import drive
drive.mount('/content/drive')
模型訓練:
將雲端硬碟掛載好之後,我們就可以開始訓練模型了。後面文章會再補充模型的介紹以及模型的堆疊。在訓練好模型之後我們會將模型儲存到雲端硬碟,方便下次直接使用模型。
import os
os.chdir('/content/pytorch-CycleGAN-and-pix2pix/')
!git clone https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix
!pip install -r requirements.txt
!bash ./datasets/download_pix2pix_dataset.sh edges2handbag #<-edges2handbag要加s,但系統會偵測他為禁止單字所以不能打出來
! python train.py --dataroot ./datasets/edges2handbag --name ro_train --model pix2pix --lr_policy step --n_epochs -99 --direction AtoB #<-edges2handbag要加s,但系統會偵測他為禁止單字所以不能打出來
訓練完成後成果圖:
如果訓練模型上遇到甚麼問題或是error的話歡迎丟到留言區討論喔!
文章主題一覽:
粗體字為額外更新的文章。